import os

from langchain_openai import ChatOpenAI

class ArkModel:
    def __init__(self, **kwargs):
        self.model = ChatOpenAI(
            openai_api_base=os.getenv('ARK_BASE_URL'),
            openai_api_key=os.getenv('ARK_BASE_KEY'),
            model_name=os.getenv('ARK_MODEL_NAME'),
            **kwargs
        )

